Using matrix form to remove a for loop MATLAB